ds and algo in python